# 引入模块os操作文件与文件夹相关的操作
import os

# 设置标志位 ：1-添加前缀  |2-删除前缀
flag = 2

# 接收用户输入的前缀
prefix = input("请输入批量重命名的前缀")

# 获取当前目录下的所有文件
file_list = os.listdir()
"""
['2.txt', '2[备份].txt', 'aa', 'bbbb', 'hm_01_体验文件操作.py', 'hm_02_主访问模式特点01.py', 'hm_03_读取函数之read.py', 'hm_04_读取函数之readlines.py', 'hm_05_读取函数之readline.py', 'hm_06_访问模式特点02.py', 'hm_07_seek函数.py', 'hm_08_文件备份.py', 'hm_09_文件和文件夹操作.py', 'hm_10_批量重命名.py', 'sound.mp3', 'sound[备份].mp3', 'test.txt', 'test1.txt', 'test200.txt', 'test[备份].txt']

"""
print(file_list)

# 循环修改文件名
for i in file_list:
    if flag == 1:
        # 添加前缀
        new_name = prefix + i
    elif flag == 2:
        # 删除前缀
        num = len('Python_')
        new_name = i[num:]
    #3. 重命名
    os.rename(i,new_name)



# # 需求1：把code文件夹所有文件重命名 Python_xxxx
# # 需求2： 删除Python_ 重命名：1， 构造条件的数据 2. 书写if
# import os
#
# # 构造条件的数据
# flag = 2
#
# # 1. 找到所有文件： 获取code文件夹的目录列表 -- listdir()
# file_list = os.listdir()
# print(file_list)
#
# # 2. 构造名字
# for i in file_list:
#     if flag == 1:
#         # new_name = 'Python_' + 原文件i
#         new_name = 'Python_' + i
#     elif flag == 2:
#         # 删除前缀
#         num = len('Python_')
#         new_name = i[num:]
# # 3. 重命名
#     os.rename(i, new_name)
